# -*- coding: utf-8 -*-

# File Name： single_gpu_train
# Description :
# Author : lirui
# create_date： 2022/6/3
# Change Activity:
from abc import ABC

from ml.dl import select_device
from ml.dl.train.base_trainer import BaseTrainer


class SingleGPUTrain(BaseTrainer, ABC):
    def __init__(self, train_cfg, model):
        super().__init__(train_cfg, model)
        self.device = select_device(train_cfg.device)
